Where do I place the return of a function in JAVA